tensor([[0.7400, 0.7400, 0.7400, 0.7400, 0.7400],
        [0.2600, 0.2600, 0.2600, 0.2600, 0.2600]], grad_fn=<RoundBackward1>)